package idv.arthur.work;

import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.google.gson.Gson;

/**
 * Servlet implementation class DelMRitem
 */
public class QueryReq extends HttpServlet {
	private static final long serialVersionUID = 1L;

	public QueryReq() {
		super();
		// TODO Auto-generated constructor stub
	}

	protected void doGet(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {
		doPost(request, response);
	}

	protected void doPost(HttpServletRequest request,
			HttpServletResponse response) throws ServletException, IOException {

		request.setCharacterEncoding("UTF-8");

		Widgets wid = new Widgets();
		Connection conn = null;
		PreparedStatement ps = null;
		ResultSet rs = null;
		String strSQL = "SELECT COUNT(*) FROM req";
		ResultTableSet resultTableSet = new ResultTableSet();
		String dataType = request.getParameter("getDataType");
		ArrayList<InfoStruc> infoStrucs = new ArrayList<InfoStruc>();
		int page = 1;
		int total_pages = 1;
		int count = 0;
		int limit = 4;

		try {
			conn = wid.getConn(true);
			ps = conn.prepareStatement(strSQL);
			rs = ps.executeQuery();
			rs.next();
			count = rs.getInt(1);

			if (count != 0) {

				if (limit < count) {
					total_pages = (count / limit) + 1;
				}

				strSQL = "SELECT reqsn, reqid, reqname,sysFromNO FROM req ORDER BY reqname ASC";
				ps = conn.prepareStatement(strSQL);
				rs = ps.executeQuery();
				ArrayList<Rows> rows = new ArrayList<Rows>();

				while (rs.next()) {
					String reqId = rs.getString("reqid");
					if ("JSON".equalsIgnoreCase(dataType)) {
						InfoStruc infoStruc = new InfoStruc();
						infoStruc.setId(reqId);
						infoStruc.setName(rs.getString("reqname"));
						infoStrucs.add(infoStruc);
					} else if ("mapping".equalsIgnoreCase(dataType)) {
						Rows row = new Rows();
						String[] cell = null;
						row.setId(reqId);
						cell = new String[4];
						cell[1] = reqId;
						cell[2] = rs.getString("reqname");
						cell[3] = String.valueOf(rs.getInt("sysFromNO"));
						row.setCell(cell);
						rows.add(row);
					} else if ("mappingDep".equalsIgnoreCase(dataType)) {
						Rows row = new Rows();
						String[] cell = null;
						row.setId(reqId);
						cell = new String[3];
						cell[0] = reqId;
						cell[1] = rs.getString("reqname");
						cell[2] = String.valueOf(rs.getInt("sysFromNO"));
						row.setCell(cell);
						rows.add(row);
					} else {
						Rows row = new Rows();
						String[] cell = null;
						row.setId(reqId);
						cell = new String[4];
						cell[0] = reqId;
						cell[1] = rs.getString("reqname");
						cell[2] = String.valueOf(rs.getInt("reqsn"));
						cell[3] = String.valueOf(rs.getInt("sysFromNO"));
						row.setCell(cell);
						rows.add(row);
					}
				}
				resultTableSet.setRows(rows);
				resultTableSet.setPage(String.valueOf(page));
				resultTableSet.setTotal(total_pages);
				resultTableSet.setRecords(String.valueOf(count));

			} else {
				page = 0;
				total_pages = 0;
				resultTableSet.setPage(String.valueOf(page));
				resultTableSet.setTotal(total_pages);
				resultTableSet.setRecords(String.valueOf(count));
			}
		} catch (SQLException e) {
			System.out.println(wid.errProcess(e));
		} finally {
			try {
				if (rs != null) {
					rs.close();
				}
				if (ps != null) {
					ps.close();
				}
				if (conn != null) {
					conn.close();
				}
			} catch (SQLException e) {
				System.out.println(wid.errProcess(e));
			}
		}

		response.setContentType("application/json");
		response.setCharacterEncoding("UTF-8");
		if ("JSON".equalsIgnoreCase(dataType)) {
			response.getWriter().write(new Gson().toJson(infoStrucs));
		} else {
			response.getWriter().write(new Gson().toJson(resultTableSet));
		}

	}
}